import { cn } from "@/lib/utils";
import * as SelectPrimitive from "@radix-ui/react-select";
import { Check, ChevronDown, ChevronUp } from "lucide-react";
import { forwardRef, ReactNode, Ref } from "react";

interface SelectProps extends SelectPrimitive.SelectProps {
  children?: ReactNode;
  setErrorFlag?: (errorFlag: boolean) => void;
}
const Select = ({
  onOpenChange,
  setErrorFlag,
  children,
  ...props
}: SelectProps) => (
  <SelectPrimitive.Root
    onOpenChange={(open) => {
      if (onOpenChange) {
        onOpenChange(open);
      }

      if (open && setErrorFlag) {
        setErrorFlag(false); // 当选择器打开时，重置错误状态
      }
    }}
    {...props}
  >
    {children}
  </SelectPrimitive.Root>
);
Select.displayName = SelectPrimitive.Root.displayName;

const SelectGroup = SelectPrimitive.Group;

const SelectValue = SelectPrimitive.Value;

interface SelectTriggerProps extends SelectPrimitive.SelectTriggerProps {
  className?: string;
  children?: ReactNode;
  errorFlag?: boolean;
}

const SelectTrigger = forwardRef(
  (
    {
      className = "",
      children,
      errorFlag = false,
      ...props
    }: SelectTriggerProps,
    ref: Ref<HTMLButtonElement>,
  ) => (
    <SelectPrimitive.Trigger
      ref={ref}
      className={cn(
        "flex h-10 w-full items-center justify-between rounded-md border border-input bg-background px-3 py-2 text-sm ring-offset-background placeholder:text-muted-foreground focus:ring-2 focus:ring-ring focus:ring-offset-2 focus:outline-none disabled:cursor-not-allowed disabled:opacity-50 [&>span]:line-clamp-1",
        className,
        errorFlag ? "border-red-500 ring-2 ring-red-500 outline-red-500" : "",
      )}
      {...props}
    >
      {children}
      <SelectPrimitive.Icon asChild>
        <ChevronDown className="h-4 w-4 opacity-50" />
      </SelectPrimitive.Icon>
    </SelectPrimitive.Trigger>
  ),
);

SelectTrigger.displayName = SelectPrimitive.SelectTrigger.displayName;

const SelectScrollUpButton = forwardRef(
  (
    { className = "", ...props }: { className?: string },
    ref: Ref<HTMLDivElement>,
  ) => (
    <SelectPrimitive.ScrollUpButton
      ref={ref}
      className={cn(
        "flex cursor-default items-center justify-center py-1",
        className,
      )}
      {...props}
    >
      <ChevronUp className="h-4 w-4" />
    </SelectPrimitive.ScrollUpButton>
  ),
);
SelectScrollUpButton.displayName =
  SelectPrimitive.SelectScrollUpButton.displayName;

const SelectScrollDownButton = forwardRef(
  (
    { className = "", ...props }: { className?: string },
    ref: Ref<HTMLDivElement>,
  ) => (
    <SelectPrimitive.ScrollDownButton
      ref={ref}
      className={cn(
        "flex cursor-default items-center justify-center py-1",
        className,
      )}
      {...props}
    >
      <ChevronDown className="h-4 w-4" />
    </SelectPrimitive.ScrollDownButton>
  ),
);
SelectScrollDownButton.displayName =
  SelectPrimitive.SelectScrollDownButton.displayName;

interface SelectorContentProps extends SelectPrimitive.SelectContentProps {
  isLoading?: boolean;
  errorMessage?: string;
  className?: string;
  children?: ReactNode;
  length: number;
}

// portal 弹出当前dom
const SelectContent = forwardRef(
  (
    {
      className = "",
      isLoading = false,
      errorMessage,
      position = "popper",
      children,
      length,
      ...props
    }: SelectorContentProps,
    ref: Ref<HTMLDivElement>,
  ) => (
    <SelectPrimitive.Portal>
      <SelectPrimitive.Content
        position={position}
        ref={ref}
        className={cn(
          "relative z-50 max-h-96 min-w-[8rem] overflow-hidden rounded-md border bg-popover text-popover-foreground shadow-md data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=closed]:zoom-out-95 data-[state=open]:animate-in data-[state=open]:fade-in-0 data-[state=open]:zoom-in-95",
          position === "popper"
            ? "data-[side=bottom]:translate-y-1 data-[side=left]:-translate-x-1 data-[side=right]:translate-x-1 data-[side=top]:-translate-y-1"
            : "",
          className,
        )}
        {...props}
      >
        <SelectScrollUpButton />
        <SelectPrimitive.Viewport
          className={cn(
            "p-1",
            position === "popper"
              ? "h-[var(--radix-select-trigger-height)] w-full min-w-[var(--radix-select-trigger-width)]"
              : "",
          )}
        >
          {isLoading ? (
            <div className="p-2 text-sm text-muted-foreground">加载中...</div>
          ) : errorMessage ? (
            <div className="p-2 text-sm text-red-500">{errorMessage}</div>
          ) : !(length > 0) ? (
            <div className="p-2 text-sm text-muted-foreground">
              暂无可用选项
            </div>
          ) : (
            children
          )}
        </SelectPrimitive.Viewport>
        <SelectScrollDownButton />
      </SelectPrimitive.Content>
    </SelectPrimitive.Portal>
  ),
);
SelectContent.displayName = SelectPrimitive.Content.displayName;

const SelectLabel = forwardRef(
  (
    { className = "", ...props }: { className?: string },
    ref: Ref<HTMLDivElement>,
  ) => (
    <SelectPrimitive.Label
      ref={ref}
      className={cn("py-1.5 pr-2 pl-8 text-sm font-semibold", className)}
      {...props}
    />
  ),
);
SelectLabel.displayName = SelectPrimitive.Label.displayName;

const SelectItem = forwardRef(
  (
    {
      className = "",
      children,
      value = "",
      ...props
    }: { className?: string; children?: ReactNode; value?: string },
    ref: Ref<HTMLDivElement>,
  ) => (
    <SelectPrimitive.Item
      ref={ref}
      className={cn(
        "relative flex w-full cursor-default items-center rounded-sm py-1.5 pr-2 pl-8 text-sm outline-none select-none focus:bg-accent focus:text-accent-foreground data-[disabled]:pointer-events-none data-[disabled]:opacity-50",
        className,
      )}
      value={value}
      {...props}
    >
      <span className="absolute left-2 flex h-3.5 w-3.5 items-center justify-center">
        <SelectPrimitive.ItemIndicator>
          <Check className="h-4 w-4" />
        </SelectPrimitive.ItemIndicator>
      </span>
      <SelectPrimitive.ItemText>{children}</SelectPrimitive.ItemText>
    </SelectPrimitive.Item>
  ),
);
SelectItem.displayName = SelectPrimitive.Item.displayName;

const SelectSeparator = forwardRef(
  (
    { className = "", ...props }: { className?: string },
    ref: Ref<HTMLDivElement>,
  ) => (
    <SelectPrimitive.Separator
      ref={ref}
      className={cn("-mx-1 my-1 h-px bg-muted", className)}
      {...props}
    />
  ),
);
SelectSeparator.displayName = SelectPrimitive.Separator.displayName;

export {
  Select,
  SelectGroup,
  SelectValue,
  SelectTrigger,
  SelectScrollUpButton,
  SelectScrollDownButton,
  SelectContent,
  SelectLabel,
  SelectItem,
  SelectSeparator,
};

export type { SelectProps };
